transforming code into beautiful idiomatic python